package com.yunji.news.utils;

import lombok.extern.slf4j.Slf4j;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;

/**
 * 文件上传工具类
 *
 * @author yunji
 * @since 2024-11-25
 */
@Slf4j
public class FileUploadUtils {

    private static final String UPLOAD_DIR = "uploads";
    private static final List<String> ALLOWED_IMAGE_TYPES = Arrays.asList(
            "jpg", "jpeg", "png", "gif", "bmp", "webp"
    );
    private static final List<String> ALLOWED_DOCUMENT_TYPES = Arrays.asList(
            "pdf", "doc", "docx", "xls", "xlsx", "ppt", "pptx", "txt"
    );
    private static final long MAX_FILE_SIZE = 10 * 1024 * 1024; // 10MB

    /**
     * 上传文件
     *
     * @param file 上传的文件
     * @param subDir 子目录（如：images, documents, etc）
     * @return 文件相对路径
     * @throws IOException 上传失败时抛出异常
     */
    public static String uploadFile(MultipartFile file, String subDir) throws IOException {
        // 参数校验
        if (file == null || file.isEmpty()) {
            throw new IllegalArgumentException("文件不能为空");
        }
        
        if (file.getSize() > MAX_FILE_SIZE) {
            throw new IllegalArgumentException("文件大小不能超过10MB");
        }

        // 获取文件扩展名
        String originalFilename = file.getOriginalFilename();
        if (!StringUtils.hasText(originalFilename)) {
            throw new IllegalArgumentException("文件名不能为空");
        }
        
        String fileExtension = getFileExtension(originalFilename);
        if (!isAllowedFileType(fileExtension)) {
            throw new IllegalArgumentException("不支持的文件类型：" + fileExtension);
        }

        // 生成文件名：UUID + 时间戳 + 扩展名
        String timestamp = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyyMMdd_HHmmss"));
        String filename = UUID.randomUUID().toString().replace("-", "") + "_" + timestamp + "." + fileExtension;

        // 创建上传目录路径：uploads/subDir/yyyy/MM/dd
        String datePath = LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy/MM/dd"));
        String relativePath = UPLOAD_DIR + "/" + subDir + "/" + datePath + "/" + filename;
        
        // 创建绝对路径
        Path uploadPath = Paths.get(relativePath);
        Path directory = uploadPath.getParent();
        
        // 确保目录存在
        if (!Files.exists(directory)) {
            Files.createDirectories(directory);
        }

        // 保存文件
        file.transferTo(uploadPath.toFile());
        
        log.info("文件上传成功：{} -> {}", originalFilename, relativePath);
        
        // 返回用于前端访问的路径
        return "/" + relativePath.replace("\\", "/");
    }

    /**
     * 上传图片文件
     *
     * @param file 图片文件
     * @return 文件相对路径
     * @throws IOException 上传失败时抛出异常
     */
    public static String uploadImage(MultipartFile file) throws IOException {
        return uploadFile(file, "images");
    }

    /**
     * 上传文档文件
     *
     * @param file 文档文件
     * @return 文件相对路径
     * @throws IOException 上传失败时抛出异常
     */
    public static String uploadDocument(MultipartFile file) throws IOException {
        return uploadFile(file, "documents");
    }

    /**
     * 删除文件
     *
     * @param filePath 文件路径
     * @return 是否删除成功
     */
    public static boolean deleteFile(String filePath) {
        if (!StringUtils.hasText(filePath)) {
            return false;
        }
        
        try {
            // 移除开头的 "/"
            if (filePath.startsWith("/")) {
                filePath = filePath.substring(1);
            }
            
            Path path = Paths.get(filePath);
            if (Files.exists(path)) {
                Files.delete(path);
                log.info("文件删除成功：{}", filePath);
                return true;
            } else {
                log.warn("文件不存在：{}", filePath);
                return false;
            }
        } catch (IOException e) {
            log.error("文件删除失败：{}", filePath, e);
            return false;
        }
    }

    /**
     * 获取文件扩展名
     *
     * @param filename 文件名
     * @return 文件扩展名（小写）
     */
    private static String getFileExtension(String filename) {
        int lastDotIndex = filename.lastIndexOf('.');
        if (lastDotIndex == -1 || lastDotIndex == filename.length() - 1) {
            return "";
        }
        return filename.substring(lastDotIndex + 1).toLowerCase();
    }

    /**
     * 检查文件类型是否允许
     *
     * @param fileExtension 文件扩展名
     * @return 是否允许
     */
    private static boolean isAllowedFileType(String fileExtension) {
        return ALLOWED_IMAGE_TYPES.contains(fileExtension) || ALLOWED_DOCUMENT_TYPES.contains(fileExtension);
    }

    /**
     * 检查是否为图片文件
     *
     * @param fileExtension 文件扩展名
     * @return 是否为图片
     */
    public static boolean isImageFile(String fileExtension) {
        return ALLOWED_IMAGE_TYPES.contains(fileExtension.toLowerCase());
    }

    /**
     * 检查是否为文档文件
     *
     * @param fileExtension 文件扩展名
     * @return 是否为文档
     */
    public static boolean isDocumentFile(String fileExtension) {
        return ALLOWED_DOCUMENT_TYPES.contains(fileExtension.toLowerCase());
    }

    /**
     * 格式化文件大小
     *
     * @param size 文件大小（字节）
     * @return 格式化后的文件大小
     */
    public static String formatFileSize(long size) {
        if (size < 1024) {
            return size + " B";
        } else if (size < 1024 * 1024) {
            return String.format("%.1f KB", size / 1024.0);
        } else if (size < 1024 * 1024 * 1024) {
            return String.format("%.1f MB", size / (1024.0 * 1024.0));
        } else {
            return String.format("%.1f GB", size / (1024.0 * 1024.0 * 1024.0));
        }
    }
}